import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO('runs/train/exp/weights/best-365.pt')  # select your model.pt path
    #model = YOLO('runs/train/exp/weights/bad-345.pt') # select your model.pt path
    model.predict(source='dataset/images/test',
                project='dataset/images',
                name='345-result',
                save=True,
                # visualize=True # visualize model features maps
                )